import os

from zhipuai import ZhipuAI
import requests
#有额度 500张 500张结束之后再换个号
client = ZhipuAI(api_key="1434409244002bf1558679111c24cd65.9y1X4l2xZRdCBEGU") # 请填写您自己的APIKey

def generatePic(description):
    # description="过年了，画一条东方巨龙，喜庆，可爱"
    response = client.images.generations(
        model="cogview-3-plus", #填写需要调用的模型名称
        prompt=description,
    )
    # 返回图片的 URL
    url = response.data[0].url  # 这是你的图片 URL

    # try:
        # 发送 GET 请求获取图片
    img_data = requests.get(url).content
    print(requests)
    # 确保请求成功
    if requests.get(url).status_code == 200:
        # file_name = url.split("/")[-1]  # 从 URL 中提取文件名
        file_name = "bg.png"
        # 将图片保存到 img 文件夹
        file_path = os.path.join('img', file_name)  # 在 'img' 文件夹中保存图片
        with open(file_path, 'wb') as file:
            file.write(img_data)

        print(f"图片已保存为: {file_path}")
    else:
        print(f"无法获取图片，状态码: {requests.status_codes}")
    # except Exception as e:
    #     print(f"下载图片时发生错误: {e}")

if __name__ == '__main__':
    generatePic("一幅清晨的山谷景象，阳光从云层中透射而出，照亮了一条蜿蜒的小径，两侧是茂密的森林。远处，一只巨大的恶龙正在追逐一只灵巧的小狐狸。")